import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--data_path", type=str, default="/home/jiayu/MyProject_2022/ROSE-2/")
parser.add_argument("--epochs", type=int, default=100)
parser.add_argument("--lr", type=int, default= 0.0003)
parser.add_argument("--snapshot", type=int, default= 3)
parser.add_argument("--test_step", type= int, default=1)
parser.add_argument("--ckpt_path", type=str, default="checkpoint_our/")
parser.add_argument("--batch_size", type=int, default=8)
parser.add_argument("--no_cuda", type= bool, default=False)
parser.add_argument("--n_gpu", type=int, default=2)



args = parser.parse_args()